home *** CD-ROM | disk | FTP | other *** search
/ io Programmo 60 / IOPROG_60.ISO / soft / c++ / gsl-1.1.1-setup.exe / {app} / src / statistics / covariance_source.c < prev    next >
Encoding:
C/C++ Source or Header  |  2000-05-04  |  3.0 KB  |  78 lines
  1. /* statistics/covar_source.c
  2.  * 
  3.  * Copyright (C) 1996, 1997, 1998, 1999, 2000 Jim Davies, Brian Gough
  4.  * 
  5.  * This program is free software; you can redistribute it and/or modify
  6.  * it under the terms of the GNU General Public License as published by
  7.  * the Free Software Foundation; either version 2 of the License, or (at
  8.  * your option) any later version.
  9.  * 
  10.  * This program is distributed in the hope that it will be useful, but
  11.  * WITHOUT ANY WARRANTY; without even the implied warranty of
  12.  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13.  * General Public License for more details.
  14.  * 
  15.  * You should have received a copy of the GNU General Public License
  16.  * along with this program; if not, write to the Free Software
  17.  * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  18.  */
  19.  
  20. static double
  21. FUNCTION(compute,covariance) (const BASE data1[], const size_t stride1,
  22.                               const BASE data2[], const size_t stride2,
  23.                               const size_t n, 
  24.                               const double mean1, const double mean2);
  25.  
  26. static double
  27. FUNCTION(compute,covariance) (const BASE data1[], const size_t stride1,
  28.                               const BASE data2[], const size_t stride2,
  29.                               const size_t n, 
  30.                               const double mean1, const double mean2)
  31. {
  32.   /* takes a dataset and finds the covariance */
  33.  
  34.   long double covariance = 0 ;
  35.  
  36.   size_t i;
  37.  
  38.   /* find the sum of the squares */
  39.   for (i = 0; i < n; i++)
  40.     {
  41.       const long double delta1 = (data1[i * stride1] - mean1);
  42.       const long double delta2 = (data2[i * stride2] - mean2);
  43.       covariance += (delta1 * delta2 - covariance) / (i + 1);
  44.     }
  45.  
  46.   return covariance ;
  47. }
  48.  
  49. double 
  50. FUNCTION(gsl_stats,covariance_m) (const BASE data1[], const size_t stride1, 
  51.                                   const BASE data2[], const size_t stride2, 
  52.                                   const size_t n, 
  53.                                   const double mean1, const double mean2)
  54. {
  55.   const double covariance = FUNCTION(compute,covariance) (data1, stride1,
  56.                                                           data2, stride2,
  57.                                                           n, 
  58.                                                           mean1, mean2);
  59.   
  60.   return covariance * ((double)n / (double)(n - 1));
  61. }
  62.  
  63. double 
  64. FUNCTION(gsl_stats,covariance) (const BASE data1[], const size_t stride1,
  65.                                 const BASE data2[], const size_t stride2,
  66.                                 const size_t n)
  67. {
  68.   const double mean1 = FUNCTION(gsl_stats,mean) (data1, stride1, n);
  69.   const double mean2 = FUNCTION(gsl_stats,mean) (data2, stride2, n);
  70.  
  71.   return FUNCTION(gsl_stats,covariance_m)(data1, stride1, 
  72.                                           data2, stride2, 
  73.                                           n, 
  74.                                           mean1, mean2);
  75. }
  76.  
  77.  
  78.